# -*- coding: utf-8 -*-
"""
author:LTH
data:
"""
import argparse

parser = argparse.ArgumentParser(description='pytorch inplementation of animegan')
# train.py配置
parser.add_argument('--project_name', type=str, default='animegan')
parser.add_argument('--train_batch_size', type=int, default=12)
parser.add_argument('--test_batch_size', type=int, default=4)
parser.add_argument('--epochs', type=int, default=200)
parser.add_argument('--lr', type=float, default=0.0001)
parser.add_argument('--use_cuda', type=bool, default=True)
parser.add_argument('--seed', type=int, default=123)
parser.add_argument('--resume', type=bool, default=True)
#windows下使用
parser.add_argument('--data_path', type=str, default='G:/datasets/cartoongan/hosoda')
parser.add_argument('--photo_path', type=str, default='G:/datasets/VOC/VOC2012/JPEGImages')
#ubuntu下使用
# parser.add_argument('--data_path', type=str, default='../cartoongan/hosoda')
# parser.add_argument('--photo_path', type=str, default='../JPEGImages')
# trainDDP.py配置
parser.add_argument('--local_rank',type=int,default=-1)
parser.add_argument('--device',default='')
parser.parse_args()
